package org.nnxy.foodboot.controller;

import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.nnxy.foodboot.entity.User;
import org.nnxy.foodboot.service.UserService;
import org.springframework.web.bind.annotation.*;

import java.util.HashMap;
import java.util.Map;

/**
 * 用户控制器
 */
@RestController
@RequestMapping("/api/user")
@RequiredArgsConstructor
@Slf4j
public class UserController {
    
    private final UserService userService;
    
    /**
     * 用户登录
     */
    @PostMapping("/login")
    public Map<String, Object> login(@RequestBody Map<String, String> loginForm) {
        String username = loginForm.get("username");
        String password = loginForm.get("password");
        
        Map<String, Object> result = new HashMap<>();
        
        try {
            // 登录并获取token
            Map<String, Object> loginResult = userService.login(username, password);
            
            // 如果返回的code不是200，表示登录失败
            Integer code = (Integer) loginResult.get("code");
            if (code != 200) {
                // 直接返回服务层的结果
                return loginResult;
            }
            
            // 登录成功，取出token和用户信息
            String token = (String) loginResult.get("token");
            User user = (User) loginResult.get("data");
            
            // 检查登录结果
            if (token == null) {
                result.put("code", 500);
                result.put("message", "系统错误：令牌生成失败");
                return result;
            }
            
            if (user == null) {
                result.put("code", 500);
                result.put("message", "系统错误：用户信息获取失败");
                return result;
            }
            
            // 确保用户对象数据完整性
            if (user.getId() == null) {
                log.error("登录成功但用户ID为空: {}", username);
                result.put("code", 500);
                result.put("message", "用户信息不完整");
                return result;
            }
            
            // 不返回密码
            user.setPassword(null);
            
            // 返回登录结果
            result.put("token", token);
            result.put("data", user); // 返回用户信息
            result.put("code", 200);
            result.put("message", "登录成功");
            return result;
        } catch (Exception e) {
            log.error("用户登录异常", e);
            result.put("code", 500);
            result.put("message", "登录失败：" + e.getMessage());
            return result;
        }
    }
    
    /**
     * 用户注册
     */
    @PostMapping("/register")
    public Map<String, Object> register(@RequestBody User user) {
        Map<String, Object> result = new HashMap<>();
        
        try {
            boolean success = userService.register(user);
            
            if (success) {
                result.put("code", 200);
                result.put("message", "注册成功");
            } else {
                result.put("code", 500);
                result.put("message", "注册失败");
            }
        } catch (Exception e) {
            log.error("用户注册异常", e);
            result.put("code", 500);
            result.put("message", "注册失败: " + e.getMessage());
        }
        
        return result;
    }
    
    /**
     * 获取用户信息
     */
    @GetMapping("/info")
    public Map<String, Object> getUserInfo(@RequestParam Integer userId) {
        Map<String, Object> result = new HashMap<>();
        
        try {
            User user = userService.getById(userId);
            
            if (user == null) {
                result.put("code", 404);
                result.put("message", "用户不存在");
                return result;
            }
            
            user.setPassword(null); // 不返回密码
            
            result.put("code", 200);
            result.put("data", user);
        } catch (Exception e) {
            log.error("获取用户信息异常", e);
            result.put("code", 500);
            result.put("message", "获取用户信息失败: " + e.getMessage());
        }
        
        return result;
    }
    
    /**
     * 更新用户信息
     */
    @PutMapping("/info")
    public Map<String, Object> updateUserInfo(@RequestBody User user) {
        Map<String, Object> result = new HashMap<>();
        
        try {
            if (user.getId() == null) {
                result.put("code", 400);
                result.put("message", "用户ID不能为空");
                return result;
            }
            
            boolean success = userService.updateById(user);
            
            if (success) {
                result.put("code", 200);
                result.put("message", "更新成功");
                
                // 返回更新后的用户信息
                User updatedUser = userService.getById(user.getId());
                if (updatedUser != null) {
                    updatedUser.setPassword(null);
                    result.put("data", updatedUser);
                }
            } else {
                result.put("code", 500);
                result.put("message", "更新失败");
            }
        } catch (Exception e) {
            log.error("更新用户信息异常", e);
            result.put("code", 500);
            result.put("message", "更新用户信息失败: " + e.getMessage());
        }
        
        return result;
    }
} 